export PROJECT_ROOT=/home/ubuntu22/projects/GotenNet-main

export PYTHONPATH=$PROJECT_ROOT:$PROJECT_ROOT/gotennet:$PYTHONPATH

CUDA_VISIBLE_DEVICES=0 python gotennet/scripts/train.py \
    experiment=qm9_u0.yaml \
    datamodule.hparams.batch_size=32 \
    logger=tensorboard